Python - Rhythm Bandits